python code to extract